package network;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.SocketException;

// UDP 版本的回显服务器
public class UdpEchoServer {
    private DatagramSocket socket = null;

    // 创建 socket 对象的同时, 要让他绑定上一个具体的端口号
    // 服务器一定要关联上具体的端口号
    public UdpEchoServer(int port) throws SocketException {
        socket = new DatagramSocket(port);
    }

    public void start() throws IOException {
        System.out.println("服务器启动");
        while (true) {
            // 只要有客户端过来, 就能提供服务
            // 1. 读取客户端发来的请求
            //    receive 方法的参数是一个输出型参数, 需要先构造一个空白的对象
            DatagramPacket requestPacket = new DatagramPacket(new byte[4096], 4096);
            socket.receive(requestPacket);
            // 此时 DatagramPacket 是一个对象, 不方便进行处理, 可以把里面的数据拿出来构造成字符串
            String request = new String(requestPacket.getData(), 0,
                    requestPacket.getLength());

            // 2. 根据请求计算响应
            String response = process(request);

            // 3. 把响应写回客户端
            //    先把 packet 对象构造好
            DatagramPacket responsePacket = new DatagramPacket(response.getBytes(),
                    response.getBytes().length, requestPacket.getSocketAddress());
            socket.send(responsePacket);

            // 4. 打印这次相应的处理中间结果
            System.out.printf("[%s:%d] req: %s; resp: %s\n",
                    requestPacket.getAddress().toString(),
                    requestPacket.getPort(),
                    request,
                    response);
        }
    }

    public String process(String request) {
        return request;
    }

    public static void main(String[] args) throws IOException {
        UdpEchoServer server = new UdpEchoServer(8080);
        server.start();
    }
}
